#!/bin/bash
set -e

# 模型与输入文件路径
MODEL="mnist_model_epoch_01.onnx"
INPUTS="save_tensors_inputs_list.json"

# FP16 模式：dump 所有中间层输出
polygraphy run ${MODEL} \
  --trt \
  --save-engine=model_fp16.trt \
  --load-inputs ${INPUTS} \
  --model-type onnx \
  --fp16 \
  --trt-outputs mark all \
  --save-outputs outputs_fp16_all_layers.json \
  --silent

# FP32 模式：dump 所有中间层输出
polygraphy run ${MODEL} \
  --trt \
  --save-engine=model_fp32.trt \
  --load-inputs ${INPUTS} \
  --model-type onnx \
  --trt-outputs mark all \
  --save-outputs outputs_fp32_all_layers.json \
  --silent
